solved assignment code 321